package com.wsoft.mq.test.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * 需新增配置
 *  publisher-confirm-type: correlated
 *      (none：禁用发布确认（默认); simple：启用发布确认模式; correlated：启用发布确认模式，并支持关联回调（CorrelationData）)
 *  publisher-returns: true
 *
 * @Author z001
 * @Date 2025/1/9 11:24
 * 发布确认回调
 */
@Component
@Slf4j
public class RabbitMqCallBackConfig implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnsCallback {


    @Resource
    private RabbitTemplate rabbitTemplate;


    @PostConstruct
    public void init(){
        //注入
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
    }



    /**
     * 交换机确认回调方法
     * @param correlationData 保存回调消息的ID及相关信息  这个是在发消息的时候填写的,未填写就会为null
     * @param b true 代表交换机收到消息, false表示交换机未收到消息
     * @param s 失败的原因,如果交换机收到消息则为空,未收到消息则返回失败的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean b, String s) {
        String  id = correlationData != null?correlationData.getId():"";
        if(b){
            log.info("交换机收到消息id为:{}的消息",id);
        }else{
            log.info("交换机还未收到id为{}的消息,原因为:{}",id,s);
        }
    }

    /**
     * 回退函数
     * 当消息传递过程中不可达 目的地的时候  将消息返回给生产者
     * 只有消息未成功投递的时候,才会进这个方法
     * @param
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error("消息{},被退回,消息为{},路由键为{}",returned.getMessage(),returned.getReplyText(),returned.getRoutingKey());
    }


}
